去噪论文阅读-8

Transfer Learning from Synthetic to Real-Noise Denoising with Adaptive Instance Normalization

之前的有些想法模块与本文撞车了。

数据:

训练集:

  • 高斯去噪模型的数据:For the Gaussian denoiser, the training images are obtained from DIV2K [43] and BSD400 [36], and noisy image is generated by AWGN model.
  • SN 生成噪声数据:We achieve pairs of SN images and noise-free images from Waterloo dataset [34] with heteroscedastic Gaussian noise model and simulating in-camera pipelines.
  • RN 真实噪声数据:trained with SIDD training set.

Method:

基于CBDNet,采用了noise level的预测形式,在UNet上加入了adaptive instance normalization 模块,模块的输入是noise level进行 $\gamma; \beta$ 的预测。AIN细节如下:
image|487x363,50%

train noise level estimator + SN的时候,使用的是俩个loss
image|503x51,75%
image|494x128,75%
$\hat{\sigma}_i(y_s)$ 是噪声图的noise level, $\sigma_i(y_s)$ 是干净图的。
train RN 的时候 (transfer learning)只使用了L1 loss。

Pipeline:

文章主要测试了模型在:AWGN,真实噪声上的效果。对于高斯去噪,train在高斯噪声模型生成的数据上。对于真实噪声文章采用了transfer learning 的方法,在生成数据上先训练,然后固定noise level estimator, SN 的Conv Layers,只训练AIN模块。
image|552x313